# -*- coding: utf-8 -*-
print('test')

from py2neo import Graph, Node, Relationship

# 连接neo4j数据库，输入地址、用户名、密码
#graph = Graph("http://127.0.0.1:7474", auth=("neo4j", "password"))
graph = Graph("bolt://127.0.0.1:7687", auth=("neo4j", "password"))

graph.delete_all() #清除neo4j中原有的结点等所有信息

# 创建结点
node1 = Node(name = '张三')   #该结点名字是张三，也是它的属性
node2 = Node('company',name = '东风科技公司')       #该结点语义类型是major，结点名字是东风科技公司，也是它的属性
node3 = Node('person',name = '李四')          #该结点语义类型是person，结点名字是李四，也是它的属性
#给节点node1增加类型
node1.add_label('person')
#给结点node1 添加一个属性 age
node1['age'] = 40
#给结点node2 添加一个属性 industry
node2['industry'] = '计算机'
#给结点node3 添加一个属性 sex
node3['sex'] = '男'
#把结点实例化 在Neo4j中显示出来
graph.create(node1)
graph.create(node2)
graph.create(node3)
# 创建关系
maojor = Relationship(node1, '实控人', node2)
friends = Relationship(node1, '事业伙伴', node3)
maojor1 = Relationship(node3, 'CFO', node2)
#把关系实例化 在Neo4j中显示出来
graph.create(maojor)
graph.create(maojor1)
graph.create(friends)


from pprint import pprint
from paddlenlp import Taskflow

ie= Taskflow('information_extraction')
schema = {'竞赛名称': ['主办方', '承办方', '已举办次数']} # Define the schema for relation extraction
ie.set_schema(schema) # Reset schema
pprint(ie('2022语言与智能技术竞赛由中国中文信息学会和中国计算机学会联合主办，百度公司、中国中文信息学会评测工作委员会和中国计算机学会自然语言处理专委会承办，已连续举办4届，成为全球最热门的中文NLP赛事之一。'))


import jiagu

# 吻别是由张学友演唱的一首歌曲。
# 《盗墓笔记》是2014年欢瑞世纪影视传媒股份有限公司出品的一部网络季播剧，改编自南派三叔所著的同名小说，由郑保瑞和罗永昌联合导演，李易峰、杨洋、唐嫣、刘天佐、张智尧、魏巍等主演。

text = '姚明1980年9月12日出生于上海市徐汇区，祖籍江苏省苏州市吴江区震泽镇，前中国职业篮球运动员，司职中锋，现任中职联公司董事长兼总经理。'
knowledge = jiagu.knowledge(text)
print(knowledge)